package com.code.origin.utils.encryption;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

/**
 * 
 * MD5工具类，提供字符串MD5加密（校验）、文件MD5值获取（校验）功能。
 * 
 * 
 */
public class MD5Utils {
  /**
   * 16进制字符集
   * 
   */
  private static final char HEX_DIGITS[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F' };

  /**
   * 指定算法为MD5的MessageDigest
   * 
   */
  private static MessageDigest messageDigest = null;

  /**
   * 初始化messageDigest的加密算法为MD5
   * 
   */
  static {
    try {
      messageDigest = MessageDigest.getInstance("MD5");
    } catch (NoSuchAlgorithmException e) {
      e.printStackTrace();
    }
  }

  /**
   * 获取文件的MD5值
   * 
   * @param file
   *          目标文件
   * @return MD5字符串
   */
  public static String getFileMD5String(File file) {
    String ret = "";
    FileInputStream in = null;
    FileChannel ch = null;

    try {
      in = new FileInputStream(file);
      ch = in.getChannel();
      ByteBuffer byteBuffer = ch.map(FileChannel.MapMode.READ_ONLY, 0, file.length());
      messageDigest.update(byteBuffer);
      ret = bytesToHex(messageDigest.digest());
    } catch (IOException e) {
      e.printStackTrace();
    } finally {
      if (in != null) {
        try {
          in.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }

      if (ch != null) {
        try {
          ch.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
    }

    return ret;
  }

  /**
   * 
   * 获取文件的MD5值
   * 
   * @param fileName
   *          目标文件的完整名称
   * @return MD5字符串
   */
  public static String getFileMD5String(String fileName) {
    return getFileMD5String(new File(fileName));
  }

  /**
   * 
   * MD5加密字符串
   * 
   * @param str
   *          目标字符串
   * @return MD5加密后的字符串
   */
  public static String getMD5String(String str) {
    return getMD5String(str.getBytes());
  }

  /**
   * 
   * MD5加密以byte数组表示的字符串
   * 
   * @param bytes
   *          目标byte数组
   * @return MD5加密后的字符串
   */
  public static String getMD5String(byte[] bytes) {
    messageDigest.update(bytes);
    return bytesToHex(messageDigest.digest());
  }

  /**
   * 
   * 校验密码与其MD5是否一致
   * 
   * @param pwd
   *          密码字符串
   * @param md5
   *          基准MD5值
   * @return 检验结果
   */
  public static boolean checkPassword(String pwd, String md5) {
    return getMD5String(pwd).equalsIgnoreCase(md5);
  }

  /**
   * 
   * 校验密码与其MD5是否一致
   * 
   * @param pwd
   *          以字符数组表示的密码
   * @param md5
   *          基准MD5值
   * @return 检验结果
   */
  public static boolean checkPassword(char[] pwd, String md5) {
    return checkPassword(new String(pwd), md5);
  }

  /**
   * 
   * 检验文件的MD5值
   * 
   * @param file
   *          目标文件
   * 
   * @param md5
   *          基准MD5值
   * 
   * @return 检验结果
   * 
   */
  public static boolean checkFileMD5(File file, String md5) {
    return getFileMD5String(file).equalsIgnoreCase(md5);
  }

  /**
   * 
   * 检验文件的MD5值
   * 
   * @param fileName
   *          目标文件的完整名称
   * 
   * @param md5
   *          基准MD5值
   * 
   * @return 检验结果
   * 
   */
  public static boolean checkFileMD5(String fileName, String md5) {
    return checkFileMD5(new File(fileName), md5);
  }

  /**
   * 
   * 将字节数组转换成16进制字符串
   * 
   * @param bytes
   *          目标字节数组
   * 
   * @return 转换结果
   * 
   */
  public static String bytesToHex(byte bytes[]) {
    return bytesToHex(bytes, 0, bytes.length);
  }

  /**
   * 
   * 将字节数组中指定区间的子数组转换成16进制字符串
   * 
   * @param bytes
   *          目标字节数组
   * 
   * @param start
   *          起始位置（包括该位置）
   * 
   * @param end
   *          结束位置（不包括该位置）
   * @return 转换结果
   * 
   */
  public static String bytesToHex(byte bytes[], int start, int end) {
    StringBuilder sb = new StringBuilder();

    for (int i = start; i < start + end; i++) {
      sb.append(byteToHex(bytes[i]));
    }

    return sb.toString();
  }

  public static String getSimenMD5(String strSrc) {
    String str1 = getMD5String(strSrc).toLowerCase();
    StringBuffer str2 = new StringBuffer("");
    str2.append(str1.substring(0, 2)).append("|").append(str1.substring(4, 10)).append("|").append(str1.substring(8, 21)).append("|").append(str1.substring(14, 19)).append("|").append(str1.substring(1, 9)).append("|").append(str1.substring(24, 27));
//     return str2.toString();
    return getMD5String(str2.toString());
  }

  /**
   * 
   * 将单个字节码转换成16进制字符串
   * 
   * @param bt
   *          目标字节
   * @return 转换结果
   */
  public static String byteToHex(byte bt) {
    return HEX_DIGITS[(bt & 0xf0) >> 4] + "" + HEX_DIGITS[bt & 0xf];
  }

  public static void main(String[] args) throws IOException {
    // long begin = System.currentTimeMillis();
    // String md5 = getFileMD5String(new File("E://ccc.rar"));
    // long end = System.currentTimeMillis();
    // System.out.println("MD5:\t" + md5 + "\nTime:\t" + (end - begin) + "ms");
    // midstr(str , 0 ,2) + midstr(str , 5 ,6) + midstr(str , 9 ,13) +
    // midstr(str , 15 ,26) + midstr(str , 2 ,8) + midstr(str , 25 ,32);
    System.out.println(getSimenMD5("dddd").toLowerCase());
  }
}
